class InputConfigs:
    """存储输入图片的默认设置"""

    def __init__(self):
        """初始化设置"""
        self.initial_image_size = (2048, 1536)
        # self.input_image_size = (512, 384)
        self.input_image_size = (200, 200)
        # self.input_image_size = (1024, 768)
        # self.batch_size = 16
        self.batch_size = 32
        self.base_dir = '/home/leo/Downloads/MVS pictures/TF2.4_mine pictures'

    def get_input_paras_level_l(self):
        return self.get_input_paras(True)

    def get_input_paras_level_h(self):
        return self.get_input_paras(False)

    def get_input_paras(self, if_data_level_l=False):
        pipe_image_size = self.initial_image_size

        if if_data_level_l:
            self.input_image_size = (512, 384)
            self.batch_size = 16
            pipe_image_size = self.input_image_size
            self.initial_image_size = self.input_image_size
            print("apply old method")
        else:
            print("apply new method")
        return self.input_image_size, pipe_image_size, self.initial_image_size, self.batch_size

class ModelConfigs:
    """存储模型相关的默认设置"""

    def __init__(self):
        self.num_classes = 3    # 需要分类的类别数
        self.base_lr = 1e-3     # 基础学习率
        self.base_epochs = 60   # 基础epoch轮数
